[BAEKJOON] 10999. 구간 합 구하기 2

Posted on December 20, 2020

문제 :

어떤 N개의 수가 주어져 있다. 그런데 중간에 수의 변경이 빈번히 일어나고 그 중간에 어떤 부분의 합을 구하려 한다. 만약에 1,2,3,4,5 라는 수가 있고, 3번째부터 4번째 수에 6을 더하면 1, 2, 9, 10, 5가 되고, 여기서 2번째부터 5번째까지 합을 구하라고 한다면 26을 출력하면 되는 것이다. 그리고 그 상태에서 1번째부터 3번째 수에 2를 빼고 2번째부터 5번째까지 합을 구하라고 한다면 22가 될 것이다.

입력 :

첫째 줄에 수의 개수 N(1 ≤ N ≤ 1,000,000)과 M(1 ≤ M ≤ 10,000), K(1 ≤ K ≤ 10,000) 가 주어진다. M은 수의 변경이 일어나는 횟수이고, K는 구간의 합을 구하는 횟수이다. 그리고 둘째 줄부터 N+1번째 줄까지 N개의 수가 주어진다. 그리고 N+2번째 줄부터 N+M+K+1번째 줄까지 세 개의 정수 a, b, c 또는 a, b, c, d가 주어지는데, a가 1인 경우 b번째 수부터 c번째 수에 d를 더하고, a가 2인 경우에는 b번째 수부터 c번째 수의 합을 구하여 출력하면 된다.

입력으로 주어지는 모든 수는 -263보다 크거나 같고, 263-1보다 작거나 같은 정수이다.

출력 :

입력으로 주어지는 모든 수는 -263보다 크거나 같고, 263-1보다 작거나 같은 정수이다.

풀이 :

세그먼트 트리 구현에 조금 익숙해지자는 마음으로 세그먼트 트리 문제를 풀어보았다. 문제는 간단히 정리하자면 구간 합 세그먼트 트리를 구현하는데 구간 값을 업데이트 할 수 있도록 구현해야했다.

처음 구현에는 단순히 구간 합을 구하는 방식으로 구간 합을 차례로 업데이트 해나가는 방식을 사용했다. 하지만 이런 방식은 모든 세부 트리 항목을 모두 업데이트 해줘야 하기 때문에 NlogN의 많은 시간이 소요되었다. 그래서 단순히 구간합을 구하고 구간 노드들을 차례로 업데이트 해주는 방식은 시간초과 결과가 나왔다.

코드 :

코드 보기/접기

#include <iostream>

#define MAX 1000000
#define ll long long

using namespace std;
int n;
ll segment[MAX * 4], *input;

ll initSegmentTree(int idx, int left, int right) {
    if (left == right) return segment[idx] = input[left];

    int mid = (left + right) / 2;
    return segment[idx] = initSegmentTree(idx * 2, left, mid) + initSegmentTree(idx * 2 + 1, mid + 1, right);
}

ll findSegmentSum(int idx, int left, int right, int curleft, int curright) {
    if (left <= curleft && curright <= right) return segment[idx];
    if (curright < left || right < curleft) return 0;

    int mid = (curleft + curright) / 2;
    return findSegmentSum(idx * 2, left, right, curleft, mid) +
           findSegmentSum(idx * 2 + 1, left, right, mid + 1, curright);
}

ll updateSegment(int idx, int value, int left, int right, int curleft, int curright) {
    if (left <= curleft && curright <= right && curleft == curright) return segment[idx] += value;
    if (curright < left || right < curleft) return segment[idx];

    int mid = (curleft + curright) / 2;
    return segment[idx] = updateSegment(idx * 2, value, left, right, curleft, mid) +
                          updateSegment(idx * 2 + 1, value, left, right, mid + 1, curright);
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int m, k, i, num, fir, sec, add;
    cin >> n >> m >> k;
    input = new ll[n];
    for (i = 0; i < n; i++)
        cin >> input[i];

    initSegmentTree(1, 0, n - 1);
    m += k;
    while (m--) {
        cin >> num;
        if (num == 1) {
            cin >> fir >> sec >> add;
            updateSegment(1, add, fir - 1, sec - 1, 0, n - 1);
        } else {
            cin >> fir >> sec;
            cout << findSegmentSum(1, fir - 1, sec - 1, 0, n - 1) << '\n';
        }
    }

    return 0;
}

Algorithm

algorithm baekjoon C++